flower code using python